package com.example.houduan.common.lang.mail;

import javax.swing.*;
import java.util.Random;

public class VerificationCodeGenerator {

    // 验证码的位数
    private static final int CODE_LENGTH = 6;

    public static void main(String[] args) {
        // 生成验证码
        String verificationCode = generateCode(CODE_LENGTH);
        // 打印验证码
        System.out.println("生成的验证码是: " + verificationCode);
    }

    /**
     * 生成指定长度的随机验证码
     * @param length 验证码的长度
     * @return 随机验证码字符串
     */
    public static String generateCode(int length) {
        // 验证码字符集，可以根据需要添加或删除字符
        String charSet = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
        // StringBuilder用于构建最终的验证码字符串
        StringBuilder code = new StringBuilder();

        // 创建一个Random对象用于生成随机数
        Random random = new Random();

        for (int i = 0; i < length; i++) {
            // 从字符集中随机选择一个字符
            int index = random.nextInt(charSet.length());
            // 将选中的字符添加到StringBuilder中
            code.append(charSet.charAt(index));
        }

        return code.toString();
    }
}
